from model.SimpleDense import AngleSolver
import torch
from config.Params import Params

params = Params()

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

model = torch.load(params.valid_model_save_path)
model = model.to(device)

in_str = "-0.047 -0.430 0.901 0.000 0.967 -0.247 -0.068 0.000 0.252 0.868 0.428 0.000 -6.389 4.868 5.629 1.000"
in_str_ls = in_str.split()
in_float_ls = list(map(float, in_str_ls))
in_vol = torch.tensor(in_float_ls)
in_vol = in_vol.to(device)
print(in_vol)

model.eval()
with torch.no_grad():
    out = model(in_vol)

print(out)